package cn.itsource.system.service.impl;

import cn.itsource.system.domain.Menu;
import cn.itsource.system.mapper.MenuMapper;
import cn.itsource.system.service.IMenuService;
import cn.itsource.basic.base.service.impl.BaseServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author huang
 * @since 2022-07-14
 */
@Service
public class MenuServiceImpl extends BaseServiceImpl<Menu> implements IMenuService {

    @Autowired
    private MenuMapper menuMapper;

    @Override
    public List<Menu> queryTree() {
        //查询出所有数据
        List<Menu> allMenus = menuMapper.loadAll();

        // 把所有的数据转换成Map结构--后面查找自己的上级使用
        HashMap<Long, Menu> menuMap = new HashMap<>();
        for(Menu menu : allMenus){
            menuMap.put(menu.getId(), menu);
        }
        //准备要返回的无限极树，把顶级部门全放到fistMenus集合中
        List<Menu> firstMenus = new ArrayList<>();

        for (Menu menu : allMenus) {
            //1.从所有数据中找出第一层级 firstMenu   我们需要返回的数据
            if(menu.getParentId() == null || menu.getParentId().intValue() == 0 ){
                firstMenus.add(menu);//最顶层的数据
            }else{
                //这里拿到的是有爹的部门
                Menu dadMenu = menuMap.get(menu.getParentId());
                //判断爹时候存在
                if(dadMenu != null){
                    //如果存在，把自己装到爹的children里
                    dadMenu.getChildren().add(menu);
                }

            }
        }
        return firstMenus;
    }
}
